OMP_NUM_THREADS=4 python -m torch.distributed.launch --nproc_per_node=8 --master_port 18101 --use_env main_imagenet.py --model mobilenet_v2 --pretrained --epochs 300 --batch-size 256 --lr 0.045 --wd 0.00004 --lr-step-size 1 --lr-gamma 0.98 --prune --cache-dataset --method group_norm --global_pruning --soft_keeping_ratio 0.5 --pretrained --target_flops 0.15 --output-dir run/imagenet/mobilenetv2_gnorm --amp